package com.qd.panda.service.park;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.cdqidi.config.mybatis.handler.GeoPoint;
import com.cdqidi.constant.ConstantDto;
import com.cdqidi.db.BaseService;
import com.cdqidi.dto.CacheKey;
import com.cdqidi.dto.GeoPointDTO;
import com.cdqidi.excel.ExcelExport;
import com.cdqidi.excel.util.ExportPage;
import com.cdqidi.exception.ApiException;
import com.cdqidi.util.GeoUtil;
import com.cdqidi.util.RedisGeoUtil;
import com.cdqidi.util.RedisTemplateUtil;
import com.qd.common.panda.domain.entity.carpark.CarParkLocationDTO;
import com.qd.panda.mapper.park.CarParkLocationMapper;
import com.qd.panda.model.CarParkLocation;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.geo.Point;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.math.BigDecimal;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Optional;

import static java.util.stream.Collectors.toList;

/**
 * 停车场位置信息
 *
 * @author sjk
 */
@RequiredArgsConstructor
@Service
@Slf4j
public class CarParkLocationService extends BaseService<CarParkLocationMapper, CarParkLocation> {

    private static final String PREFIX_ID = "panda_car_park_location_pId_{0}";
    public static final String GEO_ID = "panda_car_park_location_geo";

    private final RedisTemplateUtil<CarParkLocation> redisTemplateUtil;
    @Resource
    private HttpServletResponse response;

    /**
     * 保存
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public CarParkLocationDTO save(CarParkLocationDTO dto) {
        checkMb(dto);
        final CarParkLocation model = dtoToModel(dto);
        try {
            if (super.save(model)) {
                buildRedis(dto.getPId(), dto.getLat(), dto.getLng());
                return modelToDto(model);
            }
            throw new ApiException("添加失败");
        } finally {
            clearCache(model);
            model.freeData();
        }
    }

    /**
     * 修改
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public CarParkLocationDTO update(CarParkLocationDTO dto) {
        checkMb(dto);
        final CarParkLocation historyModel = Optional.ofNullable(getById(dto.getPId())).orElseThrow(() -> new ApiException("ID不存在,不能更新"));
        final CarParkLocation model = dtoToModel(dto);
        try {
            if (super.updateById(model)) {
                buildRedis(dto.getPId(), dto.getLat(), dto.getLng());
                return modelToDto(model);
            }
            throw new ApiException("更新失败");
        } finally {
            clearCache(historyModel);
            historyModel.freeData();
            model.freeData();
        }
    }

    /**
     * 批量添加
     *
     * @param mList 数据库对象
     */
    @Transactional(rollbackFor = Exception.class)
    public void saveBatch(List<CarParkLocation> mList) {
        try {
            if (!super.saveBatch(mList)) {
                throw new ApiException("批量添加失败");
            }
            for (CarParkLocation carParkLocation : mList) {
                buildRedis(carParkLocation.getPId(), carParkLocation.getLat(), carParkLocation.getLng());
            }
        } finally {
            clearCaches(mList);
        }
    }


    /**
     * 批量更新
     *
     * @param mList 数据库对象
     */
    @Transactional(rollbackFor = Exception.class)
    public void updateBatchIds(List<CarParkLocation> mList) {
        final List<String> ids = mList.stream().map(CarParkLocation::getPId).collect(toList());
        final List<CarParkLocation> historyModelList = checkIds(ids);
        try {
            if (!super.updateBatchById(mList)) {
                throw new ApiException("批量更新失败");
            }
            for (CarParkLocation carParkLocation : mList) {
                buildRedis(carParkLocation.getPId(), carParkLocation.getLat(), carParkLocation.getLng());
            }
        } finally {
            clearCaches(historyModelList);
            historyModelList.clear();
            ids.clear();
        }
    }

    /**
     * 删除，单个
     *
     * @param pId 主键
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeById(String pId) {
        final CarParkLocation model = Optional.ofNullable(getById(pId)).orElseThrow(() -> new ApiException("数据不存在,不能删除"));
        try {
            if (!super.removeById(pId)) {
                throw new ApiException("删除失败");
            }
        } finally {
            clearCache(model);
            clearGeoCache(model);
            model.freeData();
        }
    }

    /**
     * 根据传入对象构建查询参数，判断是否存在
     *
     * @param dto 参数
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExists(CarParkLocationDTO dto) {
        final QueryWrapper<CarParkLocation> queryWrapper = createQueryWrapper(dto);
        final boolean exists = super.count(queryWrapper) > 0;
        queryWrapper.clear();
        return exists;
    }

    /**
     * 根据主键判断数据是否存在
     *
     * @param pId 主键
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExistsById(String pId) {
        boolean exists = false;
        if (StringUtils.hasLength(pId)) {
            final CarParkLocation model = getById(pId);
            exists = (null != model);
            if (exists) {
                model.freeData();
            }
        }
        return exists;
    }

    /**
     * 根据主键获取数据
     *
     * @param pId 主键
     * @return 单条记录
     */
    @Transactional(readOnly = true)
    public CarParkLocation getById(String pId) {
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, pId);
        return redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getById(pId), CarParkLocation.class);
    }

    /**
     * 根据主键获取数据
     *
     * @param pId 主键
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public CarParkLocationDTO getByIdDto(String pId) {
        return Optional.ofNullable(getById(pId)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录
     */
    @Transactional(readOnly = true)
    public CarParkLocation getOne(CarParkLocationDTO dto) {
        final QueryWrapper<CarParkLocation> queryWrapper = createQueryWrapper(dto);
        CarParkLocation model = super.getOne(queryWrapper);
        queryWrapper.clear();
        return model;
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public CarParkLocationDTO getOneDto(CarParkLocationDTO dto) {
        return Optional.ofNullable(getOne(dto)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据
     */
    @Transactional(readOnly = true)
    public List<CarParkLocation> getList(CarParkLocationDTO dto) {
        final QueryWrapper<CarParkLocation> queryWrapper = createQueryWrapper(dto);
        final List<CarParkLocation> list = super.list(queryWrapper);
        queryWrapper.clear();
        return list;
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据，转换成页面对象
     */
    @Transactional(readOnly = true)
    public List<CarParkLocationDTO> getListDto(CarParkLocationDTO dto) {
        return modelToDtoList(getList(dto));
    }

    /**
     * 分页数据
     *
     * @param page 分页对象
     * @param dto  查询参数
     * @return 分页信息
     */
    @Transactional(readOnly = true)
    public IPage<CarParkLocationDTO> page(Page<CarParkLocation> page, CarParkLocationDTO dto) {
        final QueryWrapper<CarParkLocation> queryWrapper = createQueryWrapper(dto, "car.");
        final IPage<CarParkLocationDTO> iPage = baseMapper.page(page, queryWrapper);
        queryWrapper.clear();
        return iPage;
    }

    /**
     * 导出Excel
     *
     * @param query 查询条件
     * @throws IOException 异常
     */
    @Transactional(readOnly = true)
    public void export(CarParkLocationDTO query) throws IOException {
        final ExportPage<CarParkLocation, CarParkLocationDTO> exportPage = new ExportPage<>();
        final List<CarParkLocationDTO> list = exportPage.getAllForPage(page -> page(page, query).getRecords());
        try {
            ExcelExport.downloadExcel(response, CarParkLocationDTO.class, () -> list);
        } catch (IOException e) {
            log.error("导出失败", e);
            throw new ApiException("导出失败");
        }
    }


    /**
     * 清空缓存
     *
     * @param model 数据库对象
     */
    public void clearCache(CarParkLocation model) {
        if (null == model) {
            return;
        }
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getPId()));
    }

    public void clearGeoCache(CarParkLocation model) {
        redisTemplateUtil.getStringRedisTemplate().opsForGeo().remove(GEO_ID, model.getPId());
    }

    private void clearCaches(List<CarParkLocation> mList) {
        for (CarParkLocation carParkLocation : mList) {
            clearCache(carParkLocation);
        }
    }

    /**
     * 数据库对象转页面对象
     *
     * @param model 数据库对象
     * @return 页面对象
     */
    public CarParkLocationDTO modelToDto(CarParkLocation model) {
        if (null == model) {
            return null;
        }
        final CarParkLocationDTO dto = new CarParkLocationDTO();
        dto.setPId(model.getPId());
        dto.setAddress(model.getAddress());
        dto.setLat(model.getLat());
        dto.setLng(model.getLng());
        dto.setAdCode(model.getAdCode());
        dto.setLocationType(model.getLocationType());
        dto.setRegion(model.getRegion());
        dto.setRegionName(model.getRegionName());
        dto.setProvinceId(model.getProvinceId());
        dto.setProvinceName(model.getProvinceName());
        dto.setCityId(model.getCityId());
        dto.setCityName(model.getCityName());
        buildDisplay(dto);
        return dto;
    }

    /**
     * 页面对象转数据库对象
     *
     * @param dto 页面对象
     * @return 数据库对象
     */
    public CarParkLocation dtoToModel(CarParkLocationDTO dto) {
        if (null == dto) {
            return null;
        }
        final CarParkLocation model = new CarParkLocation();
        model.setPId(dto.getPId());
        model.setAddress(dto.getAddress());
        model.setLat(dto.getLat());
        model.setLng(dto.getLng());
        model.setAdCode(dto.getAdCode());
        model.setLocationType(dto.getLocationType());
        model.setRegion(dto.getRegion());
        model.setRegionName(dto.getRegionName());
        if (null == dto.getLat() || null == dto.getLng()) {
            log.error("parkId: {},lat: {},lng: {}", dto.getPId(), dto.getLat(), dto.getLat());
        } else {
            model.setLocation(new GeoPoint(dto.getLng(), dto.getLat()));
        }
        model.setProvinceId(dto.getProvinceId());
        model.setProvinceName(dto.getProvinceName());
        model.setCityId(dto.getCityId());
        model.setCityName(dto.getCityName());
        return model;
    }

    /**
     * 码表转换
     *
     * @param dto 页面对象
     */
    public void buildDisplay(CarParkLocationDTO dto) {
        String locationTypeDisplay = "";
        if (dto.getLocationType() == 5) {
            locationTypeDisplay = "腾讯地图";
        }
        if (dto.getLocationType() == 3) {
            locationTypeDisplay = "百度地图";
        }
        dto.setLocationTypeDisplay(locationTypeDisplay);
    }

    /**
     * 数据库对象转页面对象
     *
     * @param list 数据库对象
     * @return 页面对象
     */
    public List<CarParkLocationDTO> modelToDtoList(List<CarParkLocation> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::modelToDto).collect(toList());
    }

    /**
     * 页面对象转数据库对象
     *
     * @param list 页面对象
     * @return 数据库对象
     */
    public List<CarParkLocation> dtoToModelList(List<CarParkLocationDTO> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::dtoToModel).collect(toList());
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto 查询参数，不为空的属性会被自动加入
     */
    public QueryWrapper<CarParkLocation> createQueryWrapper(CarParkLocationDTO dto) {
        return createQueryWrapper(dto, "");
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto        查询参数，不为空的属性会被自动加入
     * @param tableAlias 表别名
     */
    public QueryWrapper<CarParkLocation> createQueryWrapper(CarParkLocationDTO dto, String tableAlias) {
        final QueryWrapper<CarParkLocation> queryWrapper = new QueryWrapper<>();
        if (null != dto) {
            if (StringUtils.hasLength(tableAlias)) {
                if (tableAlias.charAt(tableAlias.length() - 1) != '.') {
                    tableAlias += ".";
                }
            }
            if (StringUtils.hasLength(dto.getAddress())) {
                queryWrapper.eq(tableAlias + "address", dto.getAddress());
            }
            if (StringUtils.hasLength(dto.getAdCode())) {
                queryWrapper.eq(tableAlias + "ad_code", dto.getAdCode());
            }
            if (StringUtils.hasLength(dto.getProvinceId())) {
                queryWrapper.eq(tableAlias + "province_id", dto.getProvinceId());
            }
            if (StringUtils.hasLength(dto.getProvinceName())) {
                queryWrapper.eq(tableAlias + "province_name", dto.getProvinceName());
            }
            if (StringUtils.hasLength(dto.getCityId())) {
                queryWrapper.eq(tableAlias + "city_id", dto.getCityId());
            }
            if (StringUtils.hasLength(dto.getCityName())) {
                queryWrapper.eq(tableAlias + "city_name", dto.getCityName());
            }
        }
        return queryWrapper;
    }

    /**
     * 批量检查主键在数据库中是否存在
     *
     * @param ids 主键集合
     */
    public List<CarParkLocation> checkIds(List<String> ids) {
        if (null == ids || ids.isEmpty()) {
            throw new ApiException("ids不能为空");
        }
        final List<CarParkLocation> list = super.listByIds(ids);
        if (list.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(ids.toArray()));
        }
        if (ids.size() == list.size()) {
            return list;
        }
        List<String> dbIds = list.stream().map(CarParkLocation::getPId).collect(toList());
        final List<String> notExistsList = ids.stream().filter(id -> !dbIds.contains(id)).collect(toList());
        dbIds.clear();
        if (!notExistsList.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(notExistsList.toArray()));
        }
        return list;
    }

    public void buildRedis(String pId, BigDecimal lat, BigDecimal lng) {
        RedisGeoUtil.geoAdd(RedisTemplateUtil.buildKey(GEO_ID,true), new Point(lng.doubleValue(), lat.doubleValue()), pId);
    }

    public void checkMb(CarParkLocationDTO location) {
        if (location.getLocationType() == 5) {
            return;
        }
        if (location.getLocationType() != 3) {
            throw new ApiException("位置类型值错误");
        }
        GeoPointDTO geoPointDTO = GeoUtil.convertBD09ToGCJ02(location.getLat().doubleValue(), location.getLng().doubleValue());
        location.setLat(geoPointDTO.getLat());
        location.setLng(geoPointDTO.getLng());
        location.setLocationType(5);
    }
}